package com.allwees.bs.c.common.aop.exception;

import com.allwees.bs.core.modelbase.constant.ErrorMessageMap;
import com.allwees.bs.core.modelbase.vo.R;
import com.allwees.core.common.exception.BaseException;
import com.allwees.core.common.i18n.MessageSourceUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Profile;
import org.springframework.http.HttpStatus;
import org.springframework.web.HttpRequestMethodNotSupportedException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.servlet.NoHandlerFoundException;

/**
 * 2020.11.25
 * 目前针对服务器的返回，已发客户端并未做任何的特殊处理，只处理了 R.OK 正常处理，其它的错误，返回给用户提示。
 * 所以目前的捕获反馈给用户会不友好，这里需要改进一下 CODE，将不正常的显示过滤到客户端友好的提示。
 * 未来，将严格按照 allwees-code 项目的规范，去重构整个 CODE。
 */
@Profile("prod")
@Slf4j
@RestControllerAdvice
public class ProdExceptionHandler {
    /**
     * 处理所有不可知的异常
     */
    @ExceptionHandler(Exception.class)
    public R handleException(Exception e) {
        log.error("===>UnkownException:{}", e.getMessage(), e);
        return R.error(MessageSourceUtil.getMessage("client.normal.error"));
    }

    @ExceptionHandler(BaseException.class)
    public R baseException(BaseException e) {
        log.error("===>BaseException:{}", e.getMessage(), e);
        return R.error(e.getCode(), MessageSourceUtil.getMessageOrDefault(
                "result.code.(" + ErrorMessageMap.get(e.getMsg()) + ")", e.getMsg()));
    }

    @ExceptionHandler(NoHandlerFoundException.class)
    public R noHandlerFoundException(NoHandlerFoundException e) {
        log.error("===>NoHandlerFoundException:{}", e.getMessage(), e);
        return R.error(HttpStatus.NOT_FOUND.value(),
                MessageSourceUtil.getMessage("http." + HttpStatus.NOT_FOUND.value()));
    }


    @ExceptionHandler({HttpRequestMethodNotSupportedException.class})
    public R methodNotSupportedException(HttpRequestMethodNotSupportedException e) {
        log.error("===>RequestMethodNotSupportedException:{}", e.getMessage(), e);
        return R.error(HttpStatus.METHOD_NOT_ALLOWED.value(),
                MessageSourceUtil.getMessage("http." + HttpStatus.METHOD_NOT_ALLOWED.value()));
    }
}
